package com.topnet.leaderhelp.utils.xfyun;

import android.content.Context;
import android.os.Bundle;
import android.util.Log;
import android.widget.Toast;

import com.blankj.utilcode.util.LogUtils;
import com.iflytek.cloud.ErrorCode;
import com.iflytek.cloud.InitListener;
import com.iflytek.cloud.RecognizerListener;
import com.iflytek.cloud.RecognizerResult;
import com.iflytek.cloud.SpeechConstant;
import com.iflytek.cloud.SpeechError;
import com.iflytek.cloud.SpeechRecognizer;
import com.iflytek.cloud.ui.RecognizerDialog;
import com.iflytek.cloud.ui.RecognizerDialogListener;
import com.topnet.leaderhelp.R;
//import com.topnet.leaderhelp.utils.xfyun.msc.RecognizerDialog;

import org.json.JSONException;
import org.json.JSONObject;

import java.util.HashMap;
import java.util.LinkedHashMap;

/**
 * @Description: 科大讯飞 语音识别
 * @Author: 下雨了
 * @CreateDate: 2022-02-22 9:43
 * @UpdateUser: 更新者
 * @UpdateDate: 2022-02-22 9:43
 * @UpdateRemark: 更新说明
 * @Version: 1.0
 */
public class XFApiUtils {
    private static final String TAG = XFApiUtils.class.getSimpleName();

    public static XFApiUtils instance;
    private Context mContent;

    public static XFApiUtils getInstance(Context context) {
        if (instance == null) {
            instance = new XFApiUtils(context);
        }
        return instance;
    }

    public XFApiUtils(Context context) {
        mContent = context;
        init();
    }

    //语言 默认中文
    private LanguageEnum language = LanguageEnum.zh_cn;
    // 如果是中文 可以设置 语言区域如 普通、粤语 默认普通话
    private String accent = "mandarin";
    // 设置标点符号,设置为"0"返回结果无标点,设置为"1"返回结果有标点
    private String punctuation = "1";
    // 设置语音前端点:静音超时时间，即用户多长时间不说话则当做超时处理 默认4000
    private String voiceBOS = "10000";
    // 设置语音后端点:后端点静音检测时间，即用户停止说话多长时间内即认为不再输入， 自动停止录音 默认1000
    private String voiceEOS = "10000";


    // 语音听写对象
    private SpeechRecognizer mIat;
    // 语音听写UI
    private RecognizerDialog mIatDialog;
    // 用HashMap存储听写结果
    private HashMap<String, String> mIatResults = new LinkedHashMap<>();
    // 引擎类型 TYPE_CLOUD(云端识别) TYPE_LOCAL（本地识别）
    private String mEngineType = SpeechConstant.TYPE_CLOUD;
    // 返回结果格式
    private String resultType = "json";
    // 最后的结果
    private StringBuffer result = new StringBuffer();

    private Toast mToast;

    ResultCallback resultCallback;

    public void setResultCallback(ResultCallback resultCallback) {
        this.resultCallback = resultCallback;
    }

    // 初始化录音
    private void init() {
        // 初始化识别无UI识别对象
        // 使用SpeechRecognizer对象，可根据回调消息自定义界面；
        mIat = SpeechRecognizer.createRecognizer(mContent, mInitListener);


        // 初始化听写Dialog，如果只使用有UI听写功能，无需创建SpeechRecognizer
        // 使用UI听写功能，请根据sdk文件目录下的notice.txt,放置布局文件和图片资源
//        mIatDialog = new RecognizerDialog(mContent, mInitListener);

        // 听写Dialog 设置监听
//        mIatDialog.setListener(mRecognizerDialogListener);
        // 设置参数
        setParam();
    }

    // 设置参数
    private void setParam() {
        // 清空参数
        mIat.setParameter(SpeechConstant.PARAMS, null);
        // 设置听写引擎
        mIat.setParameter(SpeechConstant.ENGINE_TYPE, mEngineType);
        // 设置返回结果格式
        mIat.setParameter(SpeechConstant.RESULT_TYPE, resultType);
        if (language == LanguageEnum.zh_cn) {
            // 设置语言
            mIat.setParameter(SpeechConstant.LANGUAGE, language.name());
            // 设置语言区域 默认普通话
            mIat.setParameter(SpeechConstant.ACCENT, accent);
        } else {
            mIat.setParameter(SpeechConstant.LANGUAGE, language.name());
        }
        // 设置语音前端点:静音超时时间，即用户多长时间不说话则当做超时处理
        mIat.setParameter(SpeechConstant.VAD_BOS, voiceBOS);

        // 设置语音后端点:后端点静音检测时间，即用户停止说话多长时间内即认为不再输入， 自动停止录音
        mIat.setParameter(SpeechConstant.VAD_EOS, voiceEOS);

        mIat.setParameter("dwa", "wpgs");

        // 设置标点符号,设置为"0"返回结果无标点,设置为"1"返回结果有标点
        mIat.setParameter(SpeechConstant.ASR_PTT, punctuation);
        // 设置音频保存路径，保存音频格式支持pcm、wav.
        mIat.setParameter(SpeechConstant.AUDIO_FORMAT, "wav");
        mIat.setParameter(SpeechConstant.ASR_AUDIO_PATH, mContent.getExternalFilesDir("msc").getAbsolutePath() + "/iat.wav");
    }

    /**
     * 初始化监听器。
     */
    private final InitListener mInitListener = new InitListener() {
        @Override
        public void onInit(int code) {
            Log.d(TAG, "SpeechRecognizer init() code = " + code);
            if (code != ErrorCode.SUCCESS) {
                showTip("语音识别 初始化失败，错误码：" + code + ",请联系管理员");
            }
        }
    };




    //无UI的听写

    private final RecognizerListener mRecognizerListener = new RecognizerListener() {
        @Override
        public void onVolumeChanged(int i, byte[] bytes) {

//            LogUtils.i("--onVolumeChanged--"+bytes.);
        }

        @Override
        public void onBeginOfSpeech() {
            LogUtils.i("--onBeginOfSpeech--");

        }

        @Override
        public void onEndOfSpeech() {
            LogUtils.i("--onEndOfSpeech--");
            showTip("--onEndOfSpeech--录音已结束");
        }

        @Override
        public void onResult(RecognizerResult recognizerResult, boolean b) {

            result = printResultFix(recognizerResult);
            resultCallback.callback(result.toString());
        }

        @Override
        public void onError(SpeechError speechError) {

        }

        @Override
        public void onEvent(int i, int i1, int i2, Bundle bundle) {

        }
    };

    /**
     * 听写UI监听器
     */
    private final RecognizerDialogListener mRecognizerDialogListener = new RecognizerDialogListener() {
        // 返回结果
        public void onResult(RecognizerResult results, boolean isLast) {
            Log.w(TAG, isLast + "");
            result = printResult(results);

            resultCallback.callback(result.toString());
//            if (isLast) {
//                resultCallback.callback(result.toString());
//            }
        }

        // 识别回调错误
        public void onError(SpeechError error) {
            showTip(error.getPlainDescription(true));
            resultCallback.callback(error.getPlainDescription(true));
        }

    };



    // 读取动态修正返回结果示例代码
    private StringBuffer printResultFix(RecognizerResult results) {
        String text = JsonParser.parseIatResult(results.getResultString());

        String sn = null;
        String pgs = null;
        String rg = null;
        // 读取json结果中的sn字段
        try {
            JSONObject resultJson = new JSONObject(results.getResultString());
            sn = resultJson.optString("sn");
            pgs = resultJson.optString("pgs");
            rg = resultJson.optString("rg");
        } catch (JSONException e) {
            e.printStackTrace();
        }
        //如果pgs是rpl就在已有的结果中删除掉要覆盖的sn部分
        if (pgs.equals("rpl")) {
            String[] strings = rg.replace("[", "").replace("]", "").split(",");
            int begin = Integer.parseInt(strings[0]);
            int end = Integer.parseInt(strings[1]);
            for (int i = begin; i <= end; i++) {
                mIatResults.remove(i+"");
            }
        }

        mIatResults.put(sn, text);
        StringBuffer resultBuffer = new StringBuffer();
        for (String key : mIatResults.keySet()) {
            resultBuffer.append(mIatResults.get(key));
        }


        return resultBuffer;
    }

    /**
     * 显示结果
     * @return
     */
    private StringBuffer printResult(RecognizerResult results) {
        String text = JsonParser.parseIatResult(results.getResultString());
        String sn = null;
        // 读取json结果中的sn字段
        try {
            JSONObject resultJson = new JSONObject(results.getResultString());
            sn = resultJson.optString("sn");
        } catch (JSONException e) {
            e.printStackTrace();
        }

        mIatResults.put(sn, text);

        StringBuffer resultBuffer = new StringBuffer();
        for (String key : mIatResults.keySet()) {
            resultBuffer.append(mIatResults.get(key));
        }
        return resultBuffer;
    }

    // 显示提示
    private void showTip(final String str) {
        if (mToast != null) {
            mToast.cancel();
        }
        mToast = Toast.makeText(mContent, str, Toast.LENGTH_SHORT);
        mToast.show();
    }

    // 获取语言类型
    private LanguageEnum getLanguage(String language) {
        switch (language){
            case "zh_cn":
                return LanguageEnum.zh_cn;
            case "en_us":
                return LanguageEnum.en_us;
            case "ja_jp":
                return LanguageEnum.ja_jp;
            case "ru_ru":
                return LanguageEnum.ru_ru;
            case "es_es":
                return LanguageEnum.es_es;
            case "fr_fr":
                return LanguageEnum.fr_fr;
            case "ko_kr":
                return LanguageEnum.ko_kr;
            default:
                return LanguageEnum.zh_cn;
        }
    }

    // ======= 对外暴漏 ====================================================================================================


    // 开始录音
    public void start(ResultCallback resultCallback) {
        setResultCallback(resultCallback);
        result.setLength(0);
        mIatResults.clear();

        mIat.startListening(mRecognizerListener);
        // 显示听写对话框
//        mIatDialog.show();
        showTip(mContent.getString(R.string.text_begin));
    }

    // 结束录音
    public void stop() {
        mIat.stopListening();
        showTip("停止听写");
    }

    // 取消录音
    public void cancel() {
        mIat.cancel();
        showTip("取消听写");
    }

    // 获取语音结果
    public String getResult() {
        return result.toString();
    }

    // 语言设置 （目前支持 普通话、英语...）
    public void setLanguage(String language) {
        this.language = getLanguage(language);
    }

    // 区域设置 （普通话、粤语）
    public void setAccent(String accent) {
        this.accent = accent;
    }

    // 标点符号 （默认有）,设置为"0"返回结果无标点,设置为"1"返回结果有标点
    public void setPunctuation(String punctuation) {
        this.punctuation = punctuation;
    }

    // 设置语音前端点:静音超时时间(0-10000)，即用户多长时间不说话则当做超时处理
    public void setVoiceBOS(String voiceBOS) {
        this.voiceBOS = voiceBOS;
    }

    // 设置语音后端点:后端点静音检测时间(0-10000)，即用户停止说话多长时间内即认为不再输入， 自动停止录音
    public void setVoiceEOS(String voiceEOS) {
        this.voiceEOS = voiceEOS;
    }

}
